题目描述:
将一个二叉查找树按照中序遍历转换成双向链表。
样例:
给定一个二叉查找树:
4
/ \
2 5
/ \
1 3
返回 1<->2<->3<->4<->5。
思路讲解:这道题主要想考察的点就是二叉树的遍历与双向链表的建立,所以我们只需要分别解决这两个问题就好了,首先我们利用递归将二叉树的中序列表找出,然后我们再递归建立链表,这里建立链表我们需要注意的点就是链表的首节点的前节点是空的,尾节点的后节点也是空的,所以我们只需要注意这两个问题就好了。
代码详解:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
* Definition of Doubly-ListNode
* class DoublyListNode {
* public:
* int val;
* DoublyListNode *next, *prev;
* DoublyListNode(int val) {
* this->val = val;
this->prev = this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param root: The root of tree
* @return: the head of doubly list node
*/
DoublyListNode* bstToDoublyList(TreeNode* root) {
// Write your code here
if(root==NULL){
return NULL;
}
vector<int>res;
traveltreenode(root,res);
printint(res);
return buildlistnode(res,0,NULL);
}
void traveltreenode(TreeNode * head,vector<int>&res){//二叉树的中序遍历
if(head!=NULL){
traveltreenode(head->left,res);
res.push_back(head->val);
traveltreenode(head->right,res);
}
}
void printint(vector<int>res){
for(int i=0;i<res.size();i++){
cout<<res[i]<<" ";
}
cout<<endl;
}
DoublyListNode* buildlistnode(vector<int>res,int index,DoublyListNode* pre){//链表的建立
if(index==res.size()-1){//对于最后一个节点的特殊处理,也解决了递归的出口
DoublyListNode* end=new DoublyListNode();
end->val=res[index];
end->prev=pre;
end->next=NULL;
return end;
}
DoublyListNode* head=new DoublyListNode();
head->val=res[index];
head->prev=pre;
head->next=buildlistnode(res,index+1,head);
return head;
}
};